from collections import namedtuple
import torch
import sys 
sys.path.append("my_utiles")
from LossBuilder import LossBuilder

if __name__ == '__main__':
    Args = namedtuple("args", ["tasks_per_batch"])
    args = Args(tasks_per_batch=8)
    logits = torch.randn(5, 25)
    labels = torch.randn(25)
    l = LossBuilder.build_loss("My_Loss")().loss(logits, labels)
    print(l)